package org.tangjl.multidatasource.dynamic;

/**
 * 动态数据源上下文
 *
 * @author tangjialin1@tcl.com on 2025-08-26.
 */
public class DynamicDataSourceContextHolder {

    private static final ThreadLocal<DataSourceType> CONTEXT_HOLDER = new ThreadLocal<>();

    /**
     * 设置数据源的变量
     */
    public static void set(DataSourceType dsType) {
        CONTEXT_HOLDER.set(dsType);
    }

    /**
     * 获得数据源的变量
     */
    public static DataSourceType get() {
        return CONTEXT_HOLDER.get();
    }

    /**
     * 清空数据源变量
     */
    public static void clear() {
        CONTEXT_HOLDER.remove();
    }
}
